/*
 * Licensed to the Tai Ping Jin Ke
 *
 * Copyright (c) 2022 .
 * All rights reserved.
 * 项目名称：太平金科-通用能力封装层-公共组件
 * 版权说明：本软件属太平金融科技服务(上海)有限公司所有，在未获得太平金融科技服务(上海)有限公司正式授权情况下，任何企业和个人，不能获取、阅读、安装、传播本软件涉及的任何受知识产权保护的内容。
 */
package com.taipingframework.utility.thread.threadfactory;

import org.apache.commons.lang3.StringUtils;
import org.springframework.lang.NonNull;

import java.util.concurrent.ThreadFactory;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 自定义线程工厂
 */
public final class SmartThreadFactory implements ThreadFactory {

    private final String namePrefix;

    private final AtomicInteger nextId = new AtomicInteger(1);

    /**
     * 定义线程组名称，在进行 jstack 问题排查时，非常有帮助
     */
    public SmartThreadFactory(String threadGroupIdentity) {
        namePrefix = StringUtils.join(threadGroupIdentity, "-Worker-");
    }

    @Override
    public Thread newThread(@NonNull Runnable task) {
        String threadName = namePrefix + nextId.getAndIncrement();
        return new Thread(null, task, threadName, 0);
    }

}
